Prepare dataset

TODO